package org.dromara.exam.controller;

import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.dev33.satoken.annotation.SaMode;
import cn.hutool.core.util.IdUtil;
import jakarta.servlet.http.HttpServletResponse;
import jakarta.validation.constraints.NotEmpty;
import jakarta.validation.constraints.NotNull;
import lombok.RequiredArgsConstructor;
import org.dromara.common.core.domain.R;
import org.dromara.common.core.validate.AddGroup;
import org.dromara.common.core.validate.EditGroup;
import org.dromara.common.excel.utils.ExcelUtil;
import org.dromara.common.idempotent.annotation.RepeatSubmit;
import org.dromara.common.log.annotation.Log;
import org.dromara.common.log.enums.BusinessType;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.common.redis.utils.QueueUtils;
import org.dromara.common.redis.utils.RedisUtils;
import org.dromara.common.satoken.utils.LoginHelper;
import org.dromara.common.web.core.BaseController;
import org.dromara.exam.domain.bo.*;
import org.dromara.exam.domain.dto.Task;
import org.dromara.exam.domain.dto.TemplateImport;
import org.dromara.exam.domain.vo.ExamTemplateVo;
import org.dromara.exam.service.IExamRepoService;
import org.dromara.exam.service.IExamTemplateService;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.time.Duration;
import java.util.Collections;
import java.util.List;

/**
 * 题目
 *
 * @author xiyin321
 * @date 2024-04-11
 */
@Validated
@RequiredArgsConstructor
@RestController
@RequestMapping("/exam/template/")
public class ExamTemplateController extends BaseController {

    private final IExamTemplateService templateService;
    private final IExamRepoService repoService;

    /**
     * 分页查询题目
     *
     * @param bo        题目模板查询条件对象，包含筛选条件
     * @param pageQuery 分页查询参数对象，包含页码、页大小等信息
     * @return TableDataInfo<ExamTemplateVo> 分页数据对象，包含分页信息和数据列表
     */
    @SaCheckPermission(value = {"exam:template:page", "exam:template:query"}, mode = SaMode.OR)
    @GetMapping("page")
    public TableDataInfo<ExamTemplateVo> page(ExamTemplateBo bo, PageQuery pageQuery) {
        return templateService.queryPage(bo, pageQuery);
    }

    /**
     * 查询题目列表
     *
     * @param bo 题目模板查询条件对象，包含筛选条件
     * @return R<List < ExamTemplateVo>> 包含结果列表的响应对象
     */
    @SaCheckPermission(value = {"exam:template:list", "exam:template:query"}, mode = SaMode.OR)
    @GetMapping("list")
    public R<List<ExamTemplateVo>> list(ExamTemplateBo bo) {
        return R.ok(templateService.queryList(bo));
    }

    /**
     * 导出题目数据到Excel文件
     *
     * @param bo       题目模板查询条件对象
     * @param response HTTP响应对象，用于输出Excel文件流
     */
    @SaCheckPermission("exam:template:export")
    @Log(title = "题目", businessType = BusinessType.EXPORT)
    @PostMapping("export")
    public void export(ExamTemplateBo bo, HttpServletResponse response) {
        // 获取符合条件的数据列表
        List<ExamTemplateVo> list = templateService.queryList(bo);
        // 使用工具类导出Excel文件
        ExcelUtil.exportExcel(list, "题目", ExamTemplateVo.class, response);
    }

    /**
     * 获取指定题目的详细信息
     *
     * @param templateId 题目模板主键ID（路径参数）
     * @return R<ExamTemplateVo> 包含模板详细信息的响应对象
     */
    @SaCheckPermission("exam:template:query")
    @GetMapping("{templateId}")
    public R<ExamTemplateVo> getInfo(@NotNull(message = "主键不能为空")
                                     @PathVariable Long templateId) {
        return R.ok(templateService.queryById(templateId));
    }

    /**
     * 批量获取题目详细信息
     *
     * @param templateIds 题目模板主键ID数组（路径参数）
     * @return R<List < ExamTemplateVo>> 包含模板详细信息列表的响应对象
     */
    @SaCheckPermission("exam:template:query")
    @GetMapping("batch/{templateIds}")
    public R<List<ExamTemplateVo>> getInfo(@NotNull(message = "主键不能为空")
                                           @PathVariable Long[] templateIds) {
        return R.ok(templateService.queryListByIds(templateIds));
    }

    /**
     * 新增题目
     *
     * @param bo 题目模板业务对象，包含新增数据
     * @return R<Void> 操作结果响应对象
     */
    @SaCheckPermission("exam:template:add")
    @Log(title = "题目", businessType = BusinessType.INSERT)
    @RepeatSubmit()
    @PostMapping()
    public R<Void> add(@Validated(AddGroup.class) @RequestBody ExamTemplateBo bo) {
        return toAjax(templateService.insertByBo(bo));
    }

    /**
     * 通过AI生成题目数据
     *
     * @param bo 题目生成参数对象
     * @return R<String> 包含异步任务ID的响应对象
     */
    @SaCheckPermission("exam:template:add")
    @Log(title = "题目生成", businessType = BusinessType.GENCODE)
    @RepeatSubmit()
    @PostMapping("gen")
    public R<String> generateList(@Validated(AddGroup.class) @RequestBody QuestionGenerateBo bo) {
        // 生成唯一任务ID
        String taskId = "TASK_" + IdUtil.getSnowflakeNextIdStr();
        // 将任务存入Redis队列
        QueueUtils.addQueueObject("exam:gen:queue", new Task(taskId, LoginHelper.getUserId(), bo));
        // 初始化任务状态为处理中，有效期10分钟
        RedisUtils.setCacheObject("exam:gen:status:" + taskId, "processing", Duration.ofSeconds(600));
        return R.ok(taskId);
    }

    /**
     * 获取AI生成任务的状态或结果
     *
     * @param taskId 异步任务ID（路径参数）
     * @return R<Object> 包含任务状态/结果的响应对象，可能返回处理中状态、错误信息或生成结果列表
     */
    @Log(title = "获取任务状态", businessType = BusinessType.OTHER)
    @GetMapping("/gen/status/{taskId}")
    public R<Object> getGenStatus(@PathVariable String taskId) {
        Object status = RedisUtils.getCacheObject("exam:gen:status:" + taskId);
        if (status == null) {
            return R.fail("任务不存在或已过期");
        }
        if (status instanceof String && ((String) status).startsWith("error:")) {
            return R.fail(((String) status).substring(6));
        }
        if (status instanceof List) {
            return R.ok(status);
        }
        return R.ok(Collections.singletonMap("status", "processing"));
    }

    /**
     * 导入题目数据到编辑界面
     *
     * @param file 上传的Excel文件对象
     * @param bo   导入参数对象
     * @return R<List < TemplateImport>> 包含导入结果列表的响应对象
     */
    @SaCheckPermission("exam:template:import")
    @PostMapping(value = "/importEdit")
    @Log(title = "题库", businessType = BusinessType.IMPORT)
    public R<List<TemplateImport>> importData(@RequestPart("file") MultipartFile file, ExamTemplateImportBo bo) {
        bo.setFile(file);
        return R.ok(templateService.importTemplateToEdit(bo));
    }

    /**
     * 修改题目信息
     *
     * @param bo 题目模板业务对象，包含修改数据
     * @return R<Void> 操作结果响应对象
     */
    @SaCheckPermission("exam:template:edit")
    @Log(title = "题目", businessType = BusinessType.UPDATE)
    @RepeatSubmit()
    @PutMapping()
    public R<Void> edit(@Validated(EditGroup.class) @RequestBody ExamTemplateBo bo) {
        return toAjax(templateService.updateByBo(bo));
    }


    /**
     * 批量修改题目信息
     *
     * @param bo 批量更新业务对象，包含需要修改的ID列表和更新数据
     * @return R<Void> 操作结果响应对象
     */
    @SaCheckPermission("exam:template:edit")
    @Log(title = "题目", businessType = BusinessType.UPDATE)
    @RepeatSubmit()
    @PostMapping("batchEdit")
    public R<Void> batchEdit(@RequestBody ExamTemplateUpdateBo bo) {
        return toAjax(templateService.batchUpdateQuestionTemplate(bo));
    }

    /**
     * 删除指定题目
     *
     * @param templateIds 需要删除的题目模板主键数组（路径参数）
     * @return R<Void> 操作结果响应对象
     */
    @SaCheckPermission("exam:template:remove")
    @Log(title = "题目", businessType = BusinessType.DELETE)
    @DeleteMapping("{templateIds}")
    public R<Void> remove(@NotEmpty(message = "主键不能为空")
                          @PathVariable Long[] templateIds) {
        return toAjax(templateService.deleteWithValidByIds(List.of(templateIds), true));
    }
}
